

package com.fie.graphing;

import com.fie.contentpanes.TimeGraphPane;
import com.fie.parser.Parser;
import java.awt.BorderLayout;
import java.awt.Frame;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Timer;
import java.util.TimerTask;

import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.JTextField;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.ValueAxis;
import org.jfree.chart.plot.XYPlot;
import org.jfree.data.time.Millisecond;
import org.jfree.data.time.TimeSeries;
import org.jfree.data.time.TimeSeriesCollection;
import org.jfree.data.xy.XYDataset;
import org.jfree.ui.ApplicationFrame;
import org.jfree.ui.RefineryUtilities;

/**
 * A demonstration application showing a time series chart where you can dynamically add
 * (random) data by clicking on a button.
 *
 */
public class DynamicGraph extends ApplicationFrame implements ActionListener {

    /** The time series data. */
    private TimeSeries series;
    
    private Frame f;
    
    /** The most recent value added. */
    private double lastValue = 100.0;

    private Parser p;
    private JTextField tickerField ;
    private ValueAxis axis;
    
    
    private JFrame threeGraph;
    
    private final Timer timer = new Timer();
    /**
     * Constructs a new demonstration application.
     *
     * @param title  the frame title.
     */
    public DynamicGraph(final String title) {

        super(title);
        
        p = new Parser("CME");
        
        this.series = new TimeSeries("Last Price", Millisecond.class);
        
        final TimeSeriesCollection dataset = new TimeSeriesCollection(this.series);
        final JFreeChart chart = createChart(dataset);

        final ChartPanel chartPanel = new ChartPanel(chart);
        
        final JButton button = new JButton("Submit");
        button.setActionCommand("ADD_DATA");
        button.addActionListener(this);
        
        
        final JButton threeMonthButton = new JButton("3 months");
        threeMonthButton.setActionCommand("THREE_MONTHS");
        threeMonthButton.addActionListener(this);
        
        
        tickerField = new JTextField(10);

        final JPanel content = new JPanel();
        content.setLayout(null);
        
        chartPanel.setLocation(10,10);
        chartPanel.setSize(500,250);
        
        tickerField.setSize(100,20);
        tickerField.setLocation(520,20);
        
        button.setSize(100,20);
        button.setLocation(520,50);
        
        threeMonthButton.setSize(100,20);
        threeMonthButton.setLocation(520,70);
        
        content.add(chartPanel);
        content.add(button);
        content.add(threeMonthButton);
        content.add(tickerField);
       
        
        content.setOpaque(true);
        setContentPane(content);
        //start();

    }

    /**
     * Creates a sample chart.
     * 
     * @param dataset  the dataset.
     * 
     * @return A sample chart.
     */
    private JFreeChart createChart(final XYDataset dataset) {
        final JFreeChart result = ChartFactory.createTimeSeriesChart(
            "Real-Time Stock Chart", 
            "Time", 
            "Price",
            dataset, 
            true, 
            true, 
            false
        );
        final XYPlot plot = result.getXYPlot();
         axis = plot.getDomainAxis();
        axis.setAutoRange(true);
        axis.setFixedAutoRange(240000.0);  // 250 seconds
        axis = plot.getRangeAxis();
        axis.setRange(0.0, 200.0); 
        return result;
    }
    
    // ****************************************************************************
    // * JFREECHART DEVELOPER GUIDE                                               *
    // * The JFreeChart Developer Guide, written by David Gilbert, is available   *
    // * to purchase from Object Refinery Limited:                                *
    // *                                                                          *
    // * http://www.object-refinery.com/jfreechart/guide.html                     *
    // *                                                                          *
    // * Sales are used to provide funding for the JFreeChart project - please    * 
    // * support us so that we can continue developing free software.             *
    // ****************************************************************************
    
    /**
     * Handles a click on the button by adding new (random) data.
     *
     * @param e  the action event.
     */
    public void actionPerformed(final ActionEvent e) {
        if (e.getActionCommand().equals("ADD_DATA")) {
            
            //final double factor = 0.90 + 0.2 * Math.random();
            //this.lastValue = this.lastValue * factor;
            
            final Millisecond now = new Millisecond();        
            p = new Parser(tickerField.getText().toUpperCase());
            this.series.clear();
            System.out.println(p.getLast());
            axis.setRange(Float.parseFloat(p.getLast())-1,Float.parseFloat(p.getLast())+1);
            start();
         
        }
        else if(e.getActionCommand().equals("THREE_MONTHS")){
                f = new JFrame("Three Months");
                
                DrawStaticGraph iv = new DrawStaticGraph("http://chart.finance.yahoo.com/c/3m/c/"
                        + tickerField.getText().toLowerCase());
                f.setLayout(new BorderLayout());
                f.add("Center", iv);
                f.pack();
                f.setVisible(true);
        }
    }
    
    
    //THIS IS THE SCHEDULER TASK FOR THE GRAPH
    public void start() {
        timer.schedule(new TimerTask() {
            public void run() {
                playSound();               
            }
            private void playSound() {
                
               updateGraph();
               
               start();
            }
        }, 2000);
    }
    
    
    public void updateGraph(){
        
           // final double factor = 0.90 + 0.2 * Math.random();
           // this.lastValue = this.lastValue * factor;
            final Millisecond now = new Millisecond();
            p = new Parser(tickerField.getText().toUpperCase());
            System.out.println(p.getLast());
            
            this.series.add(new Millisecond(),Float.parseFloat(p.getLast()));
        
    }



}

